#pragma once

#include <iostream>
#include <cstdlib>
#include <signal.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

const char *dev_null = "/dev/null/";
const char *root = "/";

void Daemon(bool chdirec, bool isclose)
{
    // 关闭可能引起异常退出的信号
    signal(SIGCHLD, SIG_IGN);
    signal(SIGPIPE, SIG_IGN);

    //创建父进程退出，留下子进程
    if (fork() > 0)
        exit(1);

    // 创建守护进程
    setsid();

    // 是否更改工作目录
    if (chdirec)
        chdir(root);

    // 关闭终端输入输出，重定向文件
    if (isclose)
    {
        ::close(0);
        ::close(1);
        ::close(2);
    }
    else
    {
        int fd = open(dev_null, O_RDWR);
        if (fd > 0)
        {
            dup2(fd, 0);
            dup2(fd, 1);
            dup2(fd, 2);
            close(fd);
        }
    }
}